#include <iostream>
#include <string>
#include <memory>
#include <ctime>
#include "Socket.hpp"
#include "Protocol.hpp"
#include "InetAddr.hpp"

using namespace socket_ns;
using namespace protocol_ns;

void Usage(std::string proc)
{
    std::cout << "Usage:\n\t" << proc << " serverip serverport\n"
              << std::endl;
}

// ./tcp_client serverip serverport
int main(int argc, char *argv[])
{
    if (argc != 3)
    {
        Usage(argv[0]);
        exit(1);
    }
    std::string serverip = argv[1];
    uint16_t serverport = std::stoi(argv[2]);
    InetAddr serveraddr(serverip, serverport);

    Factory factory;
    std::unique_ptr<Socket> cli = std::make_unique<TcpSocket>();

    bool res = cli->BuildClientSocket(serveraddr);
    std::string inbuffer;

    while (res)
    {
        sleep(1);
        
        // 工厂创建请求
        auto req = factory.BuildRequest();

        // 对请求序列化
        std::string send_str;
        req->Serialize(&send_str);

        std::cout << "Serialize: \n"
                  << send_str << std::endl;

        // 加报头
        send_str = Encode(send_str);
        std::cout << "Encode: \n"
                  << send_str << std::endl;

        // 发送请求
        cli->Send(send_str);

        // 接收应答
        int n = cli->Recv(&inbuffer);
        if (n <= 0)
            break;

        // 去掉报头
        std::string package = Decode(inbuffer);
        if (package.empty())
            continue;

        // 收到完整报文
        // 创建应答
        auto resp = factory.BuildResponse();

        // 对应答反序列化
        resp->Deserialize(package);

        // 输出
        std::cout << resp->_result << "[" << resp->_code << "]" << std::endl;
    }
    return 0;
}